/*
Copyright 2021.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package v3

import (
	"net/http"

	"github.com/kubeshop/testkube/pkg/operator/clientset/versioned/scheme"

	"k8s.io/client-go/rest"

	testsv3 "github.com/kubeshop/testkube/api/tests/v3"
)

type TestsV3Interface interface {
	RESTClient() rest.Interface
	TestsGetter
	TestSuitesGetter
}

// TestsV3Client is used to interact with features provided by the tests.testkube.io group.
type TestsV3Client struct {
	restClient rest.Interface
}

func (c *TestsV3Client) Tests(namespace string) TestInterface {
	return newTests(c, namespace)
}

func (c *TestsV3Client) TestSuites(namespace string) TestSuiteInterface {
	return newTestSuites(c, namespace)
}

// NewForConfig creates a new TestsV3Client for the given config.
// NewForConfig is equivalent to NewForConfigAndClient(c, httpClient),
// where httpClient was generated with rest.HTTPClientFor(c).
func NewForConfig(c *rest.Config) (*TestsV3Client, error) {
	config := *c
	if err := setConfigDefaults(&config); err != nil {
		return nil, err
	}
	httpClient, err := rest.HTTPClientFor(&config)
	if err != nil {
		return nil, err
	}
	return NewForConfigAndClient(&config, httpClient)
}

// NewForConfigAndClient creates a new TestsV3Client for the given config and http client.
// Note the http client provided takes precedence over the configured transport values.
func NewForConfigAndClient(c *rest.Config, h *http.Client) (*TestsV3Client, error) {
	config := *c
	if err := setConfigDefaults(&config); err != nil {
		return nil, err
	}
	client, err := rest.RESTClientForConfigAndClient(&config, h)
	if err != nil {
		return nil, err
	}
	return &TestsV3Client{client}, nil
}

// NewForConfigOrDie creates a new TestsV3Client for the given config and
// panics if there is an error in the config.
func NewForConfigOrDie(c *rest.Config) *TestsV3Client {
	client, err := NewForConfig(c)
	if err != nil {
		panic(err)
	}
	return client
}

// New creates a new TestsV3Client for the given RESTClient.
func New(c rest.Interface) *TestsV3Client {
	return &TestsV3Client{c}
}

func setConfigDefaults(config *rest.Config) error {
	gv := testsv3.GroupVersion
	config.GroupVersion = &gv
	config.APIPath = "/apis"
	config.NegotiatedSerializer = scheme.Codecs.WithoutConversion()

	if config.UserAgent == "" {
		config.UserAgent = rest.DefaultKubernetesUserAgent()
	}

	return nil
}

// RESTClient returns a RESTClient that is used to communicate
// with API server by this client implementation.
func (c *TestsV3Client) RESTClient() rest.Interface {
	if c == nil {
		return nil
	}
	return c.restClient
}
